package com.myjpa.generate;

import org.junit.Test;

import java.util.Date;

/**
 * @author: yaoll
 * @date: 2020-09-04
 * @verison: 1.0
 */
public class OrderGenerator {

	boolean override = false;

	private GenerateContext initGenerateContext() {
		GenerateContext ctx = new GenerateContext();
		// 生成代码主目录
		ctx.setRootPath("/Users/devsaga/workspace/idea/myjpa/myjpa-querydsl-5");
		// response封装类包
		ctx.setResponsePackage("com.myjpa.common.web");
		// criteria工具类包
		ctx.setCriteriaPackage("com.myjpa.common.criteria");
		// 生成代码所在package
		ctx.setPkg("com.myjpa");
		ctx.setOverrideFile(override);
		return ctx;
	}

	@Test
	public void initOrder() {
		Generator generator = new Generator();
		GenerateContext ctx = initGenerateContext();
		ctx.setClassName("Order"); // 实体类
		ctx.setTableName("t_order"); // 表名
		ctx.setOverrideFile(true);
		ctx.addAttribute("sn", String.class);
		ctx.addAttribute("amount", Double.class);
		ctx.addAttribute("cnt", Integer.class);
		ctx.addAttribute("remark", String.class);
		ctx.addAttribute("createTime", Date.class);
		generator.generate(ctx);
	}

}
